import numpy as np

def predict_action(state: np.ndarray):
    """
    基线策略：随机动作 (Random Action)
    输入:
        state: np.ndarray of shape (b, 32, 20)
    输出:
        action: np.ndarray of shape (b, 1)，取值为 [-1, 0, 1]
    """
    b, l, c = state.shape
    assert l == 32 and c == 20

    # 在 [-1, 0, 1] 中随机选择
    action = np.random.choice([-1, 0, 1], size=b)

    return action.reshape(-1, 1)
